import argparse
import os
import sys
import torch

from torch.backends import cudnn

sys.path.append('.')
from config import cfg
from data import make_data_loader
from engine.trainer import do_train, do_train_with_center
from modeling import build_model
from layers import make_loss, make_loss_with_center
from solver import make_optimizer, make_optimizer_with_center, WarmupMultiStepLR

from utils.logger import setup_logger
# pth2 = torch.load('/disk2t/fby/new_version_add_oldversion_modify/tools/prw_47_warmup/epoch_10.pth')
#
# a2 = pth2.keys()

pth1 = torch.load('/disk2t/fby/strong/oim_out/resnet50_model_120.pth')
AAA =  pth1.state_dict()
# BBB = pth1.keys()

param_dict = torch.load("/disk2t/fby/strong/resnet50-19c8e357.pth")
torch.save(pth1.state_dict(), '/disk2t/fby/strong/oim_out/fby_resnet50_model_120.pth')

# pth3 = torch.load('/disk2t/fby/strong/yml_out/fby_resnet50_model_40.pth')
# a3 = pth3.keys()



# pth3 = torch.load('')
#
# a4 = pth3.keys()
# pth4 = torch.load('')
#
# a4 = pth4.keys()

aaa =100
